DATASET_DIRECTORY = "./dataset5400"
IMAGE_TYPE = ".png"

BATCH_SIZE = 30
RANDOM_SEED = 42

RANK_BATCH_SIZE = 5

IMAGENET_INPUT_H = 224
IMAGENET_INPUT_W = 224
IMAGENET_INPUT_D = 3
IMAGENET_MAX_PIXEL = 255.0

IMAGENET_IMAGE_SIZE = (IMAGENET_INPUT_H, IMAGENET_INPUT_W)
IMAGENET_SHAPE = (IMAGENET_INPUT_H, IMAGENET_INPUT_W, IMAGENET_INPUT_D)

IMAGENET_LIKE_DATASET = "imagenet_dataset"
LABELED_IMAGENET_LIKE_DATASET = "labeled_" + IMAGENET_LIKE_DATASET

RESNET_IMAGE_EMBEDDING_SIZE = 512
NUM_CLASSES = 5

EPOCHS = 10

SAVE_MODEL = True